package com.example.flow;

import java.util.concurrent.Flow;
import java.util.concurrent.SubmissionPublisher;
import java.util.concurrent.atomic.AtomicBoolean;

/**
 * 操作系统负责底层的 事件通知，但并不会直接 调用框架的回调函数。
 * 框架会通过操作系统提供的机制（如 I/O 多路复用）来监听事件的发生，然后框架会在适当的时候响应这些事件，并触发回调函数来处理它们。
 */
public class FlowDemoProcessor {

    public static void main(String[] args) {

        /**
         * 发布者
         */
        SubmissionPublisher<String> publisher = new SubmissionPublisher<>();

        /**
         * 订阅者
         */
        Flow.Subscriber<String> subscriber = new Flow.Subscriber<String>() {

            /**
             * 订阅关系
             */
            private Flow.Subscription subscription;

            /**
             * 订阅数量
             */
            private int count = 0;

            /**
             * 订阅关系是否完成
             * 可见性与原子性
             */
            AtomicBoolean isCompleted = new AtomicBoolean(false);

            /**
             * 回调函数：订阅关系
             * @param subscription a new subscription
             */
            @Override
            public void onSubscribe(Flow.Subscription subscription) {
                System.out.println(Thread.currentThread().getName() + "订阅关系" + subscription);
                this.subscription = subscription;
                this.subscription.request(7);
            }

            /**
             * 回调函数：数据
             * @param item the item emitted by the publisher
             */
            @Override
            public void onNext(String item) {
                if (isCompleted.get()) {
                    return;
                }
                if (count == 53) {
                    isCompleted.set(true);
                    this.subscription.cancel();
                    System.out.println("取消订阅");
                    return;
                }
                item = item + ":" + Thread.currentThread().getName();
                System.out.println(item);
                count++;
                if (count % 7 == 0 && !isCompleted.get()) {
                    this.subscription.request(7);
                }
            }

            /**
             * 回调函数：异常
             * @param throwable a throwable
             */
            @Override
            public void onError(Throwable throwable) {
                System.out.println(Thread.currentThread().getName() + "error" + throwable);
            }

            /**
             * 回调函数：完成
             */
            @Override
            public void onComplete() {
                System.out.println(Thread.currentThread().getName() + "complete");
            }
        };

        /**
         * 处理器
         */
        MyProcessor processor = new MyProcessor();

        /**
         * 发布者绑定订阅者
         * 触发回调函数onSubscribe
         */
        publisher.subscribe(processor);

        processor.subscribe(subscriber);

        /**
         * 发布数据
         */
        for (int i = 0; i < 67; i++) {
            /**
             * 触发回调函数onNext
             */
            publisher.submit("发布数据：" + i);
        }

        /**
         * 关闭发布者
         * 触发回调函数onComplete
         */
        publisher.close();

        /**
         * 休眠5秒，等待数据处理完成
         */
        try {
            Thread.sleep(5000);
        } catch (InterruptedException e) {
            System.out.println(e.getMessage());
        }

    }

}
